package com.atguigu.gulimall.order.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import javax.annotation.PostConstruct;

/**
 * Data time:2022/4/11 14:54
 * StudentID:2019112118
 * Author:hgw
 * Description:
 */
@Configuration
public class MyRabbitConfig {

    //@Autowired
    RabbitTemplate rabbitTemplate;

    @Primary
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory){
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        this.rabbitTemplate = rabbitTemplate;
        rabbitTemplate.setMessageConverter(messageConverter());
        initRabbitTemplate();
        return rabbitTemplate;
    }

    /**
     * 使用JSON序列化机制，进行消息转换
     * @return
     */
    @Bean
    public MessageConverter messageConverter(){
        return new Jackson2JsonMessageConverter();
    }


    /**
     * 定制 RabbitTemplate
     *  1、服务器收到消息就会回调
     *      1)、spring.rabbitmq.publisher-confirms=true
     *      2)、设置确认回调 ConfirmCallback
     *  2、消息正确抵达队列进行回调
     *      1)、spring.rabbitmq.publisher-returns: true
     *      2)、spring.rabbitmq.template.mandatory: true
     *      3)、设置确认回调 ReturnCallback
     *  3、消费端确认（保证每个消息被正确消费，此时才可以Broker删除这个消息）
     *      1)、默认是自动确认的，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     *          问题：
     *              1、收到消息，但没成功处理自动回复给服务器ACK，发生消息丢失。
     *              2、我们收到很多消息，自动回复给服务器ack,只有一个消息处理成功，宕机了。发生消息丢失
     *          解决：
     *              手动确认，每次处理完手动确认返回。spring.rabbitmq.listener.simple.acknowledge-mode=manual
     *              channel.basicAck(deliveryTag,false);    签收
     *              channel.basicNack(deliveryTag,false,true); 拒签
     *
     */
    //@PostConstruct  //在 MyRabbitConfig 对象创建完成以后，执行这个方法
    public void initRabbitTemplate() {
        // 设置消息确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 1、只要消息抵达Broker就会返回ack=true
             * @param correlationData 当前消息的唯一关联数据（这个是消息的唯一id）
             * @param b 消息是否成功还是失败
             * @param s 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean b, String s) {

                /**
                 * 1、做好消息确认机制（pulisher、consumer[手动ACK]）
                 * 2、每一个发送的消息都在数据库做好记录。定期将失败的消息再次发送一遍
                 */
                // 服务器收到了
                System.out.println("confirm...[correlationData:"+correlationData+"];[ack="+b+"];[cause="+s+"]");
            }
        });

        /**
         * 设置消息抵达队列的确认回调
         */
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 触发时机：只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message 投递失败的消息详细信息;
             * @param i 回复的状态码;
             * @param s 回复的文本内容;
             * @param s1    当时这个消息发送给哪个交换机
             * @param s2    当时这叫消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int i, String s, String s1, String s2) {
                // 报错了，修改数据库当前消息的状态->错误
                System.out.println("Fail Message["+message+"]==>i["+i+"]==>s["+s+"]==>s1["+s1+"]==>s2["+s2+"]");
            }
        });
    }
}
